






FROM mtpsl:v0.0.1



WORKDIR /




RUN  git clone https://gitee.com/xu-shaocong/mtpsl.git

WORKDIR  /mtpsl

#* install the repo 

RUN mkdir logs

# RUN pip33 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio==0.11.0 --extra-index-url https://download.pytorch.org/whl/cu113



RUN pip3 install pytorch-lightning
RUN pip3 install omegaconf
RUN pip3 install einops
RUN pip3 install progress 
RUN pip3 install open_clip_torch

RUN pip3 install numpy 
RUN pip3 install chardet
RUN pip3 install matplotlib 
RUN pip3 install scipy
RUN pip3 install loguru



#!=================================================================
#* not working
# pip3 install tensorboard==2.6.0
# pip3 install tensorboard-data-server==0.7.2
# pip3 install tensorboard-plugin-wit==1.6.post3
# pip3 install tensorboardX==1.8
#!=================================================================


RUN pip3 install ipython
RUN pip3 install jupyter 

#* for train
RUN pip3 install transformers
RUN pip3 install open_clip_torch


RUN pip3 install opencv-python





